import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.params.ParameterizedTest;
import org.junit.jupiter.params.provider.Arguments;
import org.junit.jupiter.params.provider.MethodSource;

import java.util.Arrays;
import java.util.stream.Stream;

/**
 * 旋转图像
 * 题目：给定一个 n×n 的二维矩阵表示一个图像。
 * 将图像顺时针旋转 90 度。
 * <p>
 * 说明：
 * 你必须在原地旋转图像，这意味着你需要直接修改输入的二维矩阵。请不要使用另一个矩阵来旋转图像。
 * <p>
 * 示例 1:
 * 给定 matrix =
 * [
 *  [1,2,3],
 *  [4,5,6],
 *  [7,8,9]
 * ],
 * <p>
 * 原地旋转输入矩阵，使其变为:
 * [
 *  [7,4,1],
 *  [8,5,2],
 *  [9,6,3]
 * ]
 * <p>
 * 示例 2:
 * 给定 matrix =
 * [
 *  [ 5, 1, 9,11],
 *  [ 2, 4, 8,10],
 *  [13, 3, 6, 7],
 *  [15,14,12,16]
 * ],
 * <p>
 * 原地旋转输入矩阵，使其变为:
 * [
 *  [15,13, 2, 5],
 *  [14, 3, 4, 1],
 *  [12, 6, 8, 9],
 *  [16, 7,10,11]
 * ]
 * <p>
 * 来源：力扣（LeetCode-48）
 * 链接：https://leetcode-cn.com/problems/rotate-image
 *
 * @author godfrey
 * @since 2020-11-11
 */
@DisplayName("旋转图像")
public class RotateImage extends BaseTest {

    static Stream<Arguments> testArguments() {
        return Stream.of(
                Arguments.arguments((Object) new int[][]{
                        {1, 2, 3},
                        {4, 5, 6},
                        {7, 8, 9}}),
                Arguments.arguments((Object) new int[][]{
                        {5, 1, 9, 11},
                        {2, 4, 8, 10},
                        {13, 3, 6, 7},
                        {15, 14, 12, 16}})
        );
    }

    @DisplayName("反转-时间复杂度O(n^2)，空间复杂度O(1)")
    @ParameterizedTest
    @MethodSource("testArguments")
    void rotate(int[][] matrix) {
        final int n = matrix.length;
        // 沿着对角线反转
        for (int i = 0; i < n; ++i) {
            for (int j = 0; j < n - i; ++j) {
                swap(matrix, i, j, n - 1 - j, n - 1 - i);
            }
        }
        // 沿着水平中线反转
        for (int i = 0; i < (n >> 1); ++i) {
            for (int j = 0; j < n; ++j) {
                swap(matrix, i, j, n - 1 - i, j);
            }
        }
        System.out.println(Arrays.deepToString(matrix));
    }

    private static void swap(final int[][] matrix, int i, int j, int p, int q) {
        int tmp = matrix[i][j];
        matrix[i][j] = matrix[p][q];
        matrix[p][q] = tmp;
    }
}
